#ATM系统

#主菜单效果
def menu():
    print("----------主菜单----------")
    print("欢迎来到ATM系统，请选择操作")
    print("\t查询余额 [输入1]\t")
    print("\t存款    [输入2]\t")
    print("\t取款    [输入3]\t")
    print("\t退出    [输入4]\t")
    m = int(input("请输入您的选择： "))
    return m

#查询余额函数
def yve():
    print("----------查询余额----------")
    print(f"{name}，您好，您的余额剩余：{money}元")

#存款函数
def save(y):
    print("----------存款----------")
    print(f"{name},您好，您存款{y}元成功")
    print(f"{name},您好，您的余额剩余：{money+y}元")

#取款函数
def get(z):
    print("----------取款----------")
    if z>money:
        print("你是一个小穷光蛋")
    else:
        print(f"{name},您好，您取款{z}元成功")
        print(f"{name},您好，您的余额剩余：{money - z}元")

boo = True
money = 5000000
name = input("请输入您的姓名： ")
while boo:
    chance = menu()
    if chance == 1:
        yve()
        continue
    elif chance == 2:
        save(int(input("请输入存款数： ")))
        continue
    elif chance == 3:
        get(int(input("请输入取款数")))
        continue
    else:
        print("已退出程序")
        boo = False